import autogen


# 使用本地模型（如 Ollama）
config_list = [
    {
        "model": "qwen-plus",  # 或你本地的其他模型
        "base_url": "https://dashscope.aliyuncs.com/compatible-mode/v1",
        "api_key": "sk-965dc39b016c49ecbe29de180f4db2b6",  # Ollama 通常不需要真实的 API key
    }
]

assistant = autogen.AssistantAgent(
    name="assistant",
    llm_config={"config_list": config_list},
)

user_proxy = autogen.UserProxyAgent(
    name="user_proxy",
    human_input_mode="NEVER",
    max_consecutive_auto_reply=1,
    code_execution_config={
        "use_docker": False,  # 关键设置：禁用 Docker
        "work_dir": "coding"  # 指定工作目录
    }
)

user_proxy.initiate_chat(
    assistant,
    message="请用中文创建一个简单的 Hello World 网页",
)